package com.woniuxy.admin.service.impl;

import cn.hutool.core.bean.BeanUtil;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.woniuxy.admin.exception.AdminException;
import com.woniuxy.admin.dto.MenuDto;
import com.woniuxy.admin.entity.Menu;
import com.woniuxy.admin.entity.RoleMenu;
import com.woniuxy.admin.mapper.MenuMapper;
import com.woniuxy.admin.mapper.RoleMenuMapper;
import com.woniuxy.admin.service.MenuService;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.woniuxy.redis.util.RedisKey;
import com.woniuxy.redis.util.RedisUtil;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;
import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;

/**
 * <p>
 *  服务实现类
 * </p>
 *
 * @author ...
 * @since 2024年05月22日
 */
@Service
public class MenuServiceImpl extends ServiceImpl<MenuMapper, Menu> implements MenuService {


    @Resource
    private RoleMenuMapper roleMenuMapper;
    @Resource
    private RedisUtil redisUtil;

    /**
     * 查询树型菜单结构
     * @return
     */
    @Override
    public List<MenuDto> loadAllMenu() {
        //从缓存里面 查出数据
        if(redisUtil.hasKey(RedisKey.cacheMenu())){
            List<MenuDto> menuDtos = (List<MenuDto>)redisUtil.getObject(RedisKey.cacheMenu());
            return menuDtos;
        }else {
            List<MenuDto>menuDtos=new ArrayList<MenuDto>();
            QueryWrapper<Menu> wrapper=new QueryWrapper<Menu>();
            wrapper.eq("parent_id",0);
            List<Menu> menus = this.list(wrapper);
            if(menus!=null||menus.size()>0){
                for (Menu menu : menus) {
                    MenuDto dto = BeanUtil.toBean(menu, MenuDto.class);
                    menuDtos.add(dto);
                    //递归查询子类
                    findChildrens(dto);
                }
            }
            //查完数据库后 存入缓存
            redisUtil.setObject(RedisKey.cacheMenu(),menuDtos);
            return menuDtos;
        }
    }

    /**
     * 根据角色Id 查询树形菜单结构
     * @param roles 角色Id集合
     * @return
     */
    @Override
    public List<MenuDto> loadAllMenuByRoleId(Integer roles) {
        List<RoleMenu> list = roleMenuMapper.selectList(new QueryWrapper<RoleMenu>().eq("role_id", roles));
        if(list==null||list.size()==0){
            throw new AdminException(10009,"角色还未拥有菜单");
        }
        List<MenuDto>menuDtos=new ArrayList<MenuDto>();
        List<Integer> menuIds = list.stream().map(l -> l.getMenuId()).collect(Collectors.toList());
        //所有菜单
        List<Menu> menus = this.listByIds(menuIds);
        //所有父菜单
        List<Menu> menus1 = menus.stream().filter(m -> m.getParentId() == 0).collect(Collectors.toList());
        if(menus1!=null||menus1.size()>0){
            for (Menu menu : menus1) {
                MenuDto dto = BeanUtil.toBean(menu, MenuDto.class);
                menuDtos.add(dto);
                //递归查询子类
                findChildrenss(dto,menuIds);
            }
        }
        return menuDtos;
    }

    /**
     * 递归查询子菜单
     * @param dto 权限菜单dto
     */
    private void findChildrens(MenuDto dto) {
        QueryWrapper<Menu>wrapper=new QueryWrapper<Menu>();
        wrapper.eq("parent_id",dto.getId());
        List<Menu> list = this.list(wrapper);
        if(list!=null||list.size()>0){
            for (Menu menu : list) {
                MenuDto dto1 = BeanUtil.toBean(menu, MenuDto.class);
                dto.getChildren().add(dto1);
                findChildrens(dto1);
            }
        }
    }

    private void findChildrenss(MenuDto dto,List<Integer>menuIds) {
        QueryWrapper<Menu>wrapper=new QueryWrapper<Menu>();
        wrapper.eq("parent_id",dto.getId());
        wrapper.in("id",menuIds);
        List<Menu> list = this.list(wrapper);
        if(list!=null||list.size()>0){
            for (Menu menu : list) {
                MenuDto dto1 = BeanUtil.toBean(menu, MenuDto.class);
                dto.getChildren().add(dto1);
                findChildrenss(dto1,menuIds);
            }
        }
    }
}
